package cn.net.sinodata.activity.service;
import org.junit.Test;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Properties;

import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.mail.*;
import javax.mail.internet.*;


public class EmailUtils {
    /**
     * Message对象将存储我们实际发送的电子邮件信息，
     * Message对象被作为一个MimeMessage对象来创建并且需要知道应当选择哪一个JavaMail session。
     */
    private MimeMessage message;

    /**
     * Session类代表JavaMail中的一个邮件会话。
     * 每一个基于JavaMail的应用程序至少有一个Session（可以有任意多的Session）。
     *
     * JavaMail需要Properties来创建一个session对象。
     * 寻找"mail.smtp.host"    属性值就是发送邮件的主机
     * 寻找"mail.smtp.auth"    身份验证，目前免费邮件服务器都需要这一项
     */
    private Session session;

    /***
     * 邮件是既可以被发送也可以被受到。JavaMail使用了两个不同的类来完成这两个功能：Transport 和 Store。
     * Transport 是用来发送信息的，而Store用来收信。对于这的教程我们只需要用到Transport对象。
     */
    private Transport transport;

    private String mailHost="";
    private String sender_username="";
    private String sender_password="";


    /*
     * 初始化方法
     */
    public EmailUtils() {
        InputStream in = null;
        try{
            in = new FileInputStream(new File("/Library/myWork/IdeaWorkspace/develop/activity/trunk/activity/activity-service/src/main/resources/MailServer.properties"));
        }catch (Exception e){
            e.printStackTrace();
        }
        Properties properties = new Properties();
        try {
            properties.load(in);
            this.mailHost = properties.getProperty("mail.smtp.host");
            this.sender_username = properties.getProperty("mail.sender.username");
            this.sender_password = properties.getProperty("mail.sender.password");
        } catch (IOException e) {
            e.printStackTrace();
        }

        session = Session.getInstance(properties);
        session.setDebug(true);//开启后有调试信息
        message = new MimeMessage(session);
    }

    /**
     * 发送邮件
     *
     * @param subject
     *            邮件主题
     * @param sendHtml
     *            邮件内容
     * @param receiveUser
     *            收件人地址
     */
    public void doSendHtmlEmail(String subject, String sendHtml,
                                String receiveUser, String filePath) {
        try {
            // 发件人
            //InternetAddress from = new InternetAddress(sender_username);
            // 下面这个是设置发送人的Nick name
            InternetAddress from = new InternetAddress(MimeUtility.encodeWord("千寻网","UTF-8", "B")+" <"+sender_username+">");
            message.setFrom(from);

            // 收件人
            InternetAddress to = new InternetAddress(receiveUser);
            message.setRecipient(Message.RecipientType.TO, to);//还可以有CC、BCC
            // 邮件主题
            message.setSubject(subject, "UTF-8");
            // 向multipart对象中添加邮件的各个部分内容，包括文本内容和附件
            Multipart multipart = new MimeMultipart();

            // 添加邮件正文
            BodyPart contentBodyPart = new MimeBodyPart();
            // 邮件内容
            contentBodyPart.setContent(sendHtml, "text/html;charset=UTF-8");
            multipart.addBodyPart(contentBodyPart);


            // 添加附件
            if (filePath != null && !"".equals(filePath)) {
                BodyPart attachmentBodyPart = new MimeBodyPart();
                // 根据附件路径获取文件,
                FileDataSource dataSource = new FileDataSource(filePath);
                attachmentBodyPart.setDataHandler(new DataHandler(dataSource));
                //MimeUtility.encodeWord可以避免文件名乱码
                attachmentBodyPart.setFileName(MimeUtility.encodeWord(dataSource.getFile().getName()));
                multipart.addBodyPart(attachmentBodyPart);
            }
            // 邮件的文本内容
            message.setContent(multipart);

            // 保存邮件
            message.saveChanges();

            transport = session.getTransport("smtp");
            // smtp验证，就是你用来发邮件的邮箱用户名密码
            transport.connect(mailHost, sender_username, sender_password);
            // 发送
            transport.sendMessage(message, message.getAllRecipients());
            System.out.println("send success!");
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            if(transport!=null){
                try {
                    transport.close();
                } catch (MessagingException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    @Test
    public void test() {
        EmailUtils emailUtils = new EmailUtils();
        emailUtils.doSendHtmlEmail("测试","无附件","529330509@qq.com","/Library/myWork/IdeaWorkspace/develop/activity/trunk/activity/activity-service/src/main/resources/MailServer.properties");
    }
}